package com.xigua._链表;

import java.util.ArrayList;
import java.util.List;
import java.util.Random;

/**
 * @author LiRongHua
 * @Title: ${file_name}
 * @Package ${package_name}
 * 给你一个单链表，随机选择链表的一个节点，并返回相应的节点值。每个节点 被选中的概率一样 。
 *实现 Solution 类：
 *Solution(ListNode head) 使用整数数组初始化对象。
 *   int getRandom() 从链表中随机选择一个节点并返回该节点的值。链表中所有节点被选中的概率相等
 * @date 2022/1/1620:03
 */
public class _382_链表随机结点 {

    class Solution {
        List<Integer> list;
        Random random = new Random();
        public Solution(ListNode head) {
            list = new ArrayList<>();
            while (head!=null){
                list.add(head.val);
                head = head.next;
            }
        }

        public int getRandom() {
            return list.get(random.nextInt(list.size()));
        }
    }


    /**
     * 蓄水池算法
     */
    class Solution2 {

        ListNode listNode;
        Random random = new Random();

        public Solution2(ListNode head) {
            this.listNode = head;
        }

        public int getRandom() {
            int i = 1;int ans = 0;
            for (ListNode node = listNode;node != null;node = node.next){
                if (random.nextInt(i) == 0){
                    ans = node.val;
                }
                i++;
            }
            return ans;
        }
    }

}
